Summary

This PR adds support for multiple external AI providers for file classification, extending beyond the existing local MobileBERT model. Users can now choose from various AI backends based on their preferences and requirements.

Supported Providers

Changes Made

New Files

• src/ai/providers/types.ts - Provider interface and shared types
• src/ai/providers/openai.ts - OpenAI provider implementation
• src/ai/providers/anthropic.ts - Anthropic Claude provider implementation
• src/ai/providers/gemini.ts - Google Gemini provider implementation
• src/ai/providers/ollama.ts - Ollama (local LLM) provider implementation
• src/ai/providers/local.ts - Wrapper for existing MobileBERT classifier
• src/ai/providers/index.ts - Provider factory and manager

Modified Files

• src/config.ts - Added AI provider configuration schema with env variable support
• src/core/analyzer.ts - Integrated provider system into file analysis
• src/cli.ts - Added new ai command for provider management
• src/index.ts - Exported new provider types and classes

New CLI Commands

# List available providers
sortora ai list

# Show current configuration
sortora ai info

# Set active provider (prompts for API key if needed)
sortora ai set openai
sortora ai set anthropic
sortora ai set gemini
sortora ai set ollama
sortora ai set local

# Test current provider
sortora ai test

# Use specific provider when scanning
sortora scan ~/Downloads --ai --provider openai

Configuration

Environment Variables

SORTORA_AI_PROVIDER=openai    # Override default provider
OPENAI_API_KEY=sk-...         # OpenAI API key
ANTHROPIC_API_KEY=sk-ant-...  # Anthropic API key
GEMINI_API_KEY=...            # Google Gemini API key
OLLAMA_HOST=http://localhost:11434  # Ollama server URL

Config File (~/.config/sortora/config.yaml)

ai:
  provider: openai
  openai:
    apiKey: sk-...
    model: gpt-4o-mini
  anthropic:
    apiKey: sk-ant-...
    model: claude-3-haiku-20240307
  gemini:
    apiKey: ...
    model: gemini-1.5-flash
  ollama:
    baseUrl: http://localhost:11434
    model: llama3.2

Implementation Details

• All providers implement the same AIProvider interface for consistent behavior
• Provider configuration supports both config file and environment variable overrides
• API keys are stored securely and only last 4 characters are displayed in UI
• Batch classification is supported with rate limiting for API providers
• Local provider wraps existing MobileBERT classifier for backward compatibility

Test Plan

• Build passes without errors
• Test sortora ai list command
• Test sortora ai info command
• Test sortora ai set <provider> for each provider
• Test sortora ai test with configured provider
• Test sortora scan --ai --provider <provider> for classification

This PR was written using Vibe Kanban